#!/usr/bin/env python3

# from enthought.mayavi import mlab
import numpy as np
from mayavi import mlab

x, y = np.ogrid[-2:2:20j, -2:2:20j]
z = x * np.exp(-x**2 - y ** 2)

pl = mlab.surf(x, y, z, warp_scale="auto")
# mlab.axes(xlabel='x', ylable='y', zlabel='z')
mlab.axes.set_xlabel='x'
mlab.axes.set_ylabel='y'
mlab.axes.set_zlabel='z'
mlab.outline(pl)
mlab.show()
